# -*- coding: utf-8 -*-
from app.model_predict import ModelPredict


class AppCore(object):

    def __init__(self):
        self.model_predct = ModelPredict()

    def predict(self, text):
        return self.model_predct.predict(text)


if __name__ == '__main__':
    import os
    os.chdir('..')

    app_core = AppCore()
    result = app_core.predict('昨天起，上海地铁3号线长江南路站、殷高西路站、江湾镇站三站进一步限流。体验发现，高峰时段排队5分钟能进站；不少乘客选择提前起床，“现在提前10到20分钟起床，即便限流也不会影响上班”。被限流的XDJMS，你们提前多久？新民网')["text"]
    print(result)
